#pragma once
#include <iostream>

//==========================================================================================
enum PRECISION_MODE
{
	PRECISION_FP32,
	PRECISION_FP16,
	PRECISION_INT8,
};



struct st_trans_model_input 
{
	std::string cfg_path;
	//std::string model_path;

	PRECISION_MODE precison = PRECISION_FP16;
	std::string gpu_model = "NVIDIA GeForce RTX 3060";

	//std::string trans_path;
};

const static int MAX_STRIDES = 3;
struct st_encrypt_info
{
	char gpu_model[100];

	int batch_size;
	int class_num;
	int input_channel;
	int image_width;
	int image_height;

	int stride_num;
	int strides[MAX_STRIDES];
	int grid[MAX_STRIDES][2];

	int engine_size = 0;
	st_encrypt_info()
	{
		memset(this, 0, sizeof(st_encrypt_info));
	}
};

